
if [ "$1" = "train" ]; then
	shift 1
	CUDA_VISIBLE_DEVICES=1 python run_cpg_task.py \
	--skip_train=False \
	--no_cuda=False \
	--task_name=tf_var_l4_d512_dp0.4 \
	--model_name=Transfm\
	$*
elif [ "$1" = "gen" ]; then
	shift 1
    python run_cpg_task.py --skip_train=True --no_cuda=True --task_name=transformer $*
elif [ "$1" = "vocab" ]; then
	shift 1
	python ./cpg/vocab.py --sent-freq-cutoff=10 $*
else
	echo "Invalid Option Selected"
fi